from BookCrossingLoader import BookCrossingLoader
from MovieLens import MovieLens


def LoadMovieLensData():
    ml = BookCrossingLoader(
        item_path='../ml-latest-small/book_crossing.csv'
    )
    print("Loading movie ratings...")
    data = ml.load_dataset()
    print("\nComputing movie popularity ranks so we can measure novelty later...")
    rankings = ml.get_popularity_ranks()
    return (ml, data, rankings)


if __name__ == '__main__':
    # ml, data, rankings = LoadMovieLensData()
    # print(data, rankings)
    # print(ml.getYears())
    bcl = BookCrossingLoader(
        item_path='../ml-latest-small/book_crossing.csv'
    )
    print(bcl.getCategory())
    # print(bcl.getSummery())

    ml = MovieLens()
    # print(ml.getGenres())
